package thread;

import java.util.concurrent.PriorityBlockingQueue;

/**
 * Created with IntelliJ IDEA.
 * Description:
 * User: 苏西西
 * Date: 2023-09-05
 * Time: 21:55
 */
//使用这个类来表示一个定时器中的任务
class MyTask implements Comparable<MyTask>{
    public MyTask(Runnable runnable, long time) {
        this.runnable = runnable;
        this.time = time;
    }

    //要执行的任务
    private Runnable runnable;
    //任务在执行的时候使用毫秒级时间戳
    private long time;

    //获取当前任务的时间
    public long getTime(){
        return time;
    }
    //执行任务
    public void run(){
        runnable.run();
    }

    @Override
    public int compareTo(MyTask o) {
        //返回小于、大于或等于0三种情况
        //this比o小 返回小于0
        //this比o大，返回大于0
        //this和o相同，返回等于0
        //当前需要实现的是队首元素是时间最小的
        return (int) (this.time-o.time);
    }
}

//咱们自己写一个简单的定时器
class MyTimer{
    //扫描线程
    private Thread t = null;
    //一个阻塞优先级队列来保存任务
    private PriorityBlockingQueue<MyTask> queue = new PriorityBlockingQueue<>();
    //构造方法里面创建线程
    public MyTimer(){
        t = new Thread(() ->{
            while(true){
                try {
                    //取出队首元素，检查队首元素是否到时间了
                    // 如果时间没到，塞回队列中
                    //如果时间到，就执行任务
                    synchronized (this) {
                        MyTask myTask = queue.take();
                        long curTime = System.currentTimeMillis();
                        if(curTime <myTask.getTime()){
                            //还没到点，先不执行
                            queue.put(myTask);
                            //在put之后进行一个wait
                                this.wait(myTask.getTime()-curTime);
                        }else{
                            //时间到了，执行任务
                            myTask.run();
                        }
                    }
                } catch (InterruptedException e) {
                    throw new RuntimeException(e);
                }
            }
        });
        t.start();
    }

    //指定两个参数
    //第一个参数是任务、内容
    //第二个参数是任务、在多少毫秒后执行
    public void schedule(Runnable runnable,long after){
        //注意这里的时间上的换算
        MyTask task = new MyTask(runnable,System.currentTimeMillis() +after);
        queue.put(task);
        synchronized (this) {
            this.notify();
        }
    }

}
public class ThreadDemo25 {
    public static void main(String[] args) {
        MyTimer myTimer = new MyTimer();
        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("任务1");
            }
        },1000);

        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("任务2");
            }
        },2000);
    }
}
